import socket

# 1、创建套接字
tcp_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
# 2、获取服务器的ip和端口
tcp_ip = input('请输入服务器的IP地址：')
tcp_port = int(input('请输入端口号：'))
# 3、连接服务器
tcp_socket.connect((tcp_ip, tcp_port))
# 4、获取要下载的文件名
download_file_name = input('请输入要下载的文件名：')
# 5、把要下载的文件名发送给服务器
tcp_socket.send(download_file_name.encode('utf-8'))
# 从服务器端接受文件数据
recv_data = tcp_socket.recv(1024)
# 7、保存接受的文件数据并存在文件中
if recv_data:
    with open('新下载：' + download_file_name, 'wb') as file:
        file.write(recv_data)
else:
    print('无法下载')
# 8、关闭套接字
tcp_socket.close()
